#ifndef _COMM_HPP_
#define _COMM_HPP_

#include <iostream>
#include <cerrno>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>

#define MAX_SIZE 4096

// 这里的PATHNAME为当前路径
#define PATHNAME "."
// 这里的PROJ_ID为随机值
#define PROJ_ID 0x66

key_t getKey()
{
    // ftok函数创建key值
    // ftok函数根据所提供的路径和值会通过算法确定一个唯一值
    // 当server和client看到同一个key值 也就意味着看到同一份共享内存
    key_t k = ftok(PATHNAME, PROJ_ID);
    if (k == -1)
    {
        std::cerr << errno << ": " << strerror(errno) << std::endl;
        exit(1);
    }
    return k;
}

int getShmHelper(key_t k, int flags)
{
    // shmget函数创建共享内存
    int shmid = shmget(k, MAX_SIZE, flags);
    if (shmid < 0)
    {
        std::cerr << errno << ": " << strerror(errno) << std::endl;
        exit(2);
    }
    return shmid;
}

int getshm(key_t k)
{
    return getShmHelper(k, IPC_CREAT);
}

int createShm(key_t k)
{
    // 创建新的shm权限为0600 只有自己有读写权限
    return getShmHelper(k, IPC_CREAT | IPC_EXCL | 0600);
}

void *attachShm(int shmid)
{
    void *mem = shmat(shmid, nullptr, 0);
    // 这里为啥要强转成longlong类型呢？
    // 因为当前OS为64位 指针占8个字节
    if ((long long)mem == -1L)
    {
        // 挂接失败
        std::cerr << "shmat: " << errno << strerror(errno) << std::endl;
        exit(3);
    }
    return mem;
}

void detachShm(void *start)
{
    if (shmdt(start) == -1)
    {
        // 去关联失败
        std::cerr << "shmdt: " << errno << strerror(errno) << std::endl;
    }
}

void delShm(int shmid)
{
    if (shmctl(shmid, IPC_RMID, nullptr) == -1)
    {
        std::cerr << "shmctl: " << errno << strerror(errno) << std::endl;
    }
}

#endif